Fixing UI for the home screen