Minor updates to main screen