added a better main screen