This looks much nicer/generic, was not aware of the collections module. Thanks!