Adding a class to the images to have them nicely in the box