Tweaked some UI stuff