Improvements to loading screen