More images needed sqauring.

Fix #40 basically