Add ability to retry the build