Added a setup.py for easy installation