write more unit tests