Added loading HUD, and fixed up some UI