Minor modifications to PSTH