- add more unit tests