Minor changes to learner