Assignment 2: Collections Solution