Jko/retries

First attempt at managing retries in the driver