diff --git a/examples/1. simple_image_classifier/train.py b/examples/1. simple_image_classifier/train.py
index 4cbf9b3..9df14ca 100644
--- a/examples/1. simple_image_classifier/train.py	
+++ b/examples/1. simple_image_classifier/train.py	
@@ -13,4 +13,4 @@
 output = np.concatenate((np.ones(500), np.zeros(500)))
 
 #Train the neural network using backpropagation
-model.train(input, output, batch_size = 16, epoch_number = 100, rate = 0.6)
+pof.train(model, input, output, batch_size = 16, epoch_number = 100, rate = 0.6)