Added color parameter to display function