Made changes to main screen