Minor updates

A small UI tweak, and a fix for #173