Update the build flow