Add a PSTH plot function