Added tests for collection()