Add remove collection functionality