Adjust the Driver

Change the design of the driver for a cleaner output