Update home screen (and fixed a bug)