Use model instead of model name
Refactor