Add check for model