Allow `collection` to be a function (#25)
Release 0.3.1