Update to latest driver code