Minor fixes to display