Make the example more flexible with configuration flags
Adds description to the model