Update spark driver to use 0.2.1