Added testing code for collections