add tests for my_collect method
add code to make a collect method