[wip] Add new data to learner model.

Will fix #338