corrections on augment input script