Add unit tests. Covert Mock model to instance.