working on the main screen