Fix setup.py for packaging