forgot the tests, here they are