A couple UI tweaks