single parameter requirement for square images

Update to accept a single dimension parameter in order to create square images.