Further work on python build script