Implemented the model registry