Update SquareBattery.java