Update calc_mean_std.py