Slight tweaks to GUI