Fixed setup.py to include module...