Fix setup.py for install with pip