Fix setup.py for installing by pip