Update to paper version, including datasets