Added first version of driver code