Add set_collection method.