changed back to square images for the assignment