Module docstrings
Refactor and add more docs
Add readme