Minor tweaks to the gui