A few UI tweaks

As mentioned in #28

Closing #28