Small update to home screen code