Small fixes in the image