Add setup.py

Add setup.py for install as a package