update demo to work with syntax highlighting in pycharm