Improve styling of image grid