Add remove method to model