functionalized scatter_demo.py
pep8 cleanup on scatter_demo.py