Add setup.py, move to module to not pollute global namespace