Improve the docs after having tested the module