DOC/CLN: Cleanups plotting.py