Revert "Revert "updated plotting.py""

This reverts commit 05a7250ee14efa9833548a6b6f3580e9110d5ad3.
Revert "Revert "Revert "updated plotting.py"""

This reverts commit 8e9c642bcbab815e30ecdb34ecf56a033dd26315.