Update setup.py
Merge pull request #3 from uzarubin/patch-3

Update setup.py