Add square battery mod