Restored old driver code to yank.py so that we can compare the two.