added some ui stuff