Updated build.py script