Updated some UI stuff