Created a setup.py to allow installation via pip