Fixed some ui stuff