diff --git a/week4/submit.py b/week4/submit.py
index 9f82d6a..5a2b513 100644
--- a/week4/submit.py
+++ b/week4/submit.py
@@ -4,7 +4,7 @@
 
 
 # code_size = 71
-# img_shape = (38, 38, 3)
+# img_shape = (38, 38, 3) or (32, 32, 3)
 def submit_autoencoder(submission, score, email, token):
     grader = grading.Grader("9TShnp1JEeeGGAoCUnhvuA")
     encoder, decoder = submission