Fix filenames in demo_2D.py