controller and driver code is updated

Updates driver code