Update 1ddrift.py