Minor tweaks to GUI