script now averages across all epochs for final figure