Small tweaks to display