Add retry logic
Bump version to 0.4.0