Add module to setup.py for installation