Adding diff_drv.py