Added some UI elements