Small tweaks to gui