added 2 more unit tests