Add .remove to model