pep8 cleanup on scatter_demo.py

Pep8 cleanup