Add timeout option support to travis_wait

If the first argument given to travis_wait looks like an integer,
treat it as a new timeout (instead of the default 20 minutes).
Also, tweak timeout message.