update setup.py as suggested in PR discussion