Implemented retry.wrap for #6, also added tests and documentaion

Great that you added tests as well! :+1: