Updated remove model.
Previously was not removing.
Very inefficient however seems to work