Update to drive system