add 2 unit tests