Updated driver to produce runnable code