Update battery_box.lua

Added Brandon's protection code