adding image and modified code for styling