Replace Set with collection_of gem