Update plotting.py

Converted some plt to ax, this fixed interference with other subplots in the same figure which had been causing them to sometimes display with wrong ylim etc.