Update plotting.py

plot on ax, not plt. This prevents spiketrain being plotted on bottom subplot in a figure with subplots.