CLN: Cleanups plotting.py