Add source files and setup.py