Updated driver script.