Update setup.py

Make setup script more sensible