Add tests for generating koans