Modified the layout to be responsive