Added test class for collection