Add setup.py and some test code