Now updated so it uses the full dataset pickles.