Added machine learner code