Added simple resizer in class_test.py. Now it resizes the image to 28x28 square by itself.