Updated driver code
Updated readme