Improvements in the setup.py script