Improvements on physics-informed models for lithium batteries