feat: Ability to set timeout