Added some UI stuff
Ignore me.