Update setup.py and docs