using the diamond dataset to show the visualization