Added setup.py script
Some minor issues