Improvements On Physics-Informed Models For Lithium Batteries