Add collection ref to models and use it to remove model from collection