fixing ui
fixing some ui elements