Update driver.py

fixed bug in unit_prop function
todo: test backtrack!!