More modifications to the main screen UI