feat: more tests!