refactored parameter name
Added __getitem__ for Collection and updated test to cover new code