add some tests for learning, still not quite working