Added test for smaller images
Code formatting