Add setup.py to allow pip install