Yep I'll add some tests.