Is this related to this PR? I am fine if the example plot looks the same but runs faster though.