Reworked learning to using prepareLearningArguments function