Minor tweaks to gui