updated driver code