Adding more unit tests for collections