Added loading screen support