Adding new files and updating setup.py