diff --git a/.gitignore b/.gitignore
index ad7ed72..afde035 100644
--- a/.gitignore
+++ b/.gitignore
@@ -80,3 +80,8 @@ target/
 
 # Mac OS-specific storage files
 .DS_Store
+
+# makefile step
+sync_collections
+resize_collections
+build_tfrecords
\ No newline at end of file
diff --git a/Makefile b/Makefile
index 990ad3d..2d9fb48 100644
--- a/Makefile
+++ b/Makefile
@@ -39,10 +39,19 @@ resize_collections: src/data/resize_collections.py
 	$(PYTHON_INTERPRETER) src/data/resize_collections.py
 	touch resize_collections
 
+## Build TFRecords
 build_tfrecords: sync_collections resize_collections src/features/build_tfrecords.py
 	$(PYTHON_INTERPRETER) src/features/build_tfrecords.py
 	touch build_tfrecords
 
+## Train model 
+train_model: build_tfrecords
+	$(PYTHON_INTERPRETER) src/models/train_model.py
+
+## test
+draft: 
+	$(PYTHON_INTERPRETER) src/models/data_loader.py
+
 
 ## Delete all compiled Python files
 clean: