Add option for task timeout