Configured setup.py so it will work for installing