Update setup.py script