Clean up Tree2DataSet for MC