Add example driver code