Closes #120
TODO add support to output in
TODO rewrite the code to be more readable, I will do this next week
As you can see in the plot, it is getting hard to distinguish the LearnerND from the Learner2D :D
