Updated run-demo.py

Demo now working for a single source.Improvements going on.