Updated collections exercise