Moar minor changes to UI