Fixed up the drive.py module. Also added unittests for it. More to be added.