I am running the demo code provided here. But I do not get the same plots with the nice coloring and hi