added some ui improvement