Add minimal support to display as graph