fixing some UI stuff