Updated build script for python