Put shape learner into a class