Update scatter_demo.py

colors added