Add a duration input to the model