Improved main screen UI