Fixed the screen code, looks like it is functioning correctly