add loading screen; clean up the UI a bit