Add ability to set model ID and description as args
Add ability to set model ID and description as args