Better code separation for image
Better styling of the rectangle