Add more files in setup.py