Making `wait` method of Driver/Session return condition at method exit.

Fixes #301.