Fix the setup.py to include the required files