add display logic