Cleanup and added small loading screen.