Added some UI