updates to dataset optimizer for ranking