Changed the UI a bit, made the screen smaller