Improves API; get model from a DIC