Adding collection handling for identity return

Adds the ability for the 'Return(identity)' overload to cope with
collections.
Adding code review changes
Merge pull request #25 from cskardon/issue-24

Adding collection handling for identity return