Minor UI changes. Adresses #7, closes #1.