some ui corrections