Loading screens and UI improvements