Some improvements to the loading screen