update setup.py
Merge pull request #6 from yannik-ammann/master

update setup.py