Update setup.py

Adding missing dependency