Updated the documentation and setup.py