Update setup.py for module refactor.