Improvements in the setup.py