Added some UI and graphical elements