Plot improvement, adding inf value.
Hopefully fixing some differences in the plots.
Ensure axis alginment.