Add query timeout