Add setup.py
Add code