Add type hints

When running mypy on the code that uses flaky, you get:

```
Skipping analyzing 'flaky': found module but no type hints or library stubs
```

[Here](https://mypy.readthedocs.io/en/latest/running_mypy.html#missing-type-hints-for-third-party-library) is how it can be fixed.