Updating setup.py so that `python setup.py build` works correctly.