Add an option to specify the timeout