Updated the build for setup.py