do some ui enhancement