Some small UI improvements. Closes #55