Update driver to work with Winston's circuit