Improves setup.py file