Add setup.py so can be installed via pip