Improving dependencies in setup.py