updating the layout to new design