Fix basic_demo.py.
Fix plotting.py.