Implement WithTimeOut parameter

Implement WithTimeout since it wasn't implemented before.