Minor UI changes

Fixes #1