Add docstrings to functions

[lucasrafaldini]

I think this package is awesome!

Maybe some docstrings would help better understanding the functions and the flow.