Update code to make it full screen