add tests

- [ ] add tests