Save the proper dataset