Cleanup square battery mod