Updated driver code. It now works!