UI tweaks
0.6.2