updated driver script