Add setup.py and test