Update setup.py
Merge pull request #2 from marcelomilo/master

Update