Add option to configure the duration