changed the pipeline to pick top k from all classes