Updated setup.py to include new module requirements