diff --git a/.gitignore b/.gitignore
index f3436fe1fd3e8a7064887098b38e50dfda48b27d..f50e97646d664513f94d61a7d165e1a497cb8bab 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,9 +1,8 @@
-*.jpg
 .DS_Store
 
 # Data
 data/*
-transfer_learning/hymenoptera_data/*
+hymenoptera_data/*
 
 # Torch model
 *.pt
diff --git a/submarine.jpg b/submarine.jpg
new file mode 100644
index 0000000000000000000000000000000000000000..6ae8cb7037013df9f8fbf70e5f17cb807876a342
Binary files /dev/null and b/submarine.jpg differ
diff --git a/tennis_ball.jpg b/tennis_ball.jpg
new file mode 100644
index 0000000000000000000000000000000000000000..b578de1cb1800a502353f5110517511c8d7ccc48
Binary files /dev/null and b/tennis_ball.jpg differ