Add docs
PEP: separate functions by two lines
Import everything at the root
Improve doc