REF/CLN: Standardized matplotlib imports
* Access objects via top level `matplotlib` interface where available
* Only utilize `matplotlib.pyplot` interface for visualization manipulation
Looks like tests are passing here so merging. Happy to follow up if needed