Added styling for image front and proper logic