Fixing some UI stuff