added some ui functionality