ui: Add a loading screen