few ui tweaks