Fix @retry docs and refactor the code

[masoud-msk]

Fixed the docs on the number of function invocation and the examples.

And as a bonus I refactored the implementation :upside_down_face: