Add tqdm for mean and std computation