Got setup.py to work