Minor changes

This pull request contains a demo.py which was adapted to match pep8. It seems that there is no nice wayto do this.