Update the layout to support full screen