Trying to fix setup.py to work with pip