Fix images and add more styling