some changes to supervised learner