UI tweaks. Fixes #121.