Put the screen render method inside the main class