Add --timeout flag