Make the setup.py more customizable