update driver script that allows for restart run