Update setup.py

Updated with the new module name