Update add_weight.lua