repaired code to include only mean and std