Update setup.py
Update documentation