Changes to UI - Looks pretty now! :dancers: