Update: driver code

Signed-off-by: Harshavardhana <fharshav@redhat.com>