- fixed some UI stuff