Added setup.py so the package can be installed using pip