Minor cleanup of scatterplot.py
Add missing imports