Improved the layout of the screen module