add some missing imports to plotting.py