fixed some ui stuff