Improved loading screen initialization