adding some ui tweaks