Adding change for setup.py