Update setup.py to match requirements elsewhere