Fix setup.py and related stuff. Thanks jezdez