UI tweaks after seeing it on my phone