Cleanup build.py