Added test for extend collection class