Minor updates to Display unit