Minor UI improvements to #10