Added stuff to drift.py...