Handle errors separately for each type of plot in timestep_diagnostics()

Handle errors separately for each type of plot in timestep_diagnostics() #834