pep8 cleanup on scatter_demo.py