diff --git a/test_model.py b/test_model.py
index efe4892..2ede54b 100644
--- a/test_model.py
+++ b/test_model.py
@@ -1,6 +1,6 @@
 """
 Simple example on how to fine tune models in Keras and how to use them.
-Part 1
+Part 2
 
 Source: https://www.guru99.com/keras-tutorial.html
 """