Made small fix in register function - if model is already in collection, remove old one, so new one can be added.