Fixed some UI Stuff