Waveform plot needs axis labels

The waveform plot added in !484 (merged) doesn't include axis labels. We should fix this.