Fixing setup.py to include all submodules