Minor update to setup.py (version = 2)