Add support for setting timeouts