Add setup.py

Add setup script.