improving the setup.py file