added some ui
added some ui