making changes for the images to work