added some ui changes