Update README.md

Author: sourface94

README.md

@@ -14,6 +14,7 @@ pip install hyperlib
 
 Creating a hyperbolic neural network using Keras:
 ```
+import tensorflow as tf
 from tensorflow import keras
 from hyperlib.nn.layers.lin_hyp import LinearHyperbolic
 from hyperlib.nn.optimizers.rsgd import RSGD
@@ -22,7 +23,7 @@ from hyperlib.manifold.poincare import Poincare
 # Create layers
 hyperbolic_layer_1 = LinearHyperbolic(32, Poincare(), 1)
 hyperbolic_layer_2 = LinearHyperbolic(32, Poincare(), 1)
-output_layer = lin_hyp.LinearHyperbolic(10, Poincare(), 1)
+output_layer = LinearHyperbolic(10, Poincare(), 1)
 
 # Create optimizer
 optimizer = RSGD(learning_rate=0.1)