Add back in the cPickle import