added some UI tweaks