Made the setup.py work properly (I think)