diff --git a/.gitignore b/.gitignore
index fef9282e..3c039038 100644
--- a/.gitignore
+++ b/.gitignore
@@ -157,5 +157,6 @@ actions-runner
 fedlab_benchmarks/algorithm/fedavg/standalone/*.txt
 fedlab_benchmarks/algorithm/fedavg/standalone/*.ipynb
 fedlab_benchmarks/algorithm/fedavg/standalone/*.sh
+tests/data/mnist
 codecov
 codecov_upload.sh
\ No newline at end of file
diff --git a/README.md b/README.md
index 28652eea..a8019a5a 100644
--- a/README.md
+++ b/README.md
@@ -2,7 +2,7 @@
 
 # FedLab: A Flexible Federated Learning Framework
 
-[![Documentation Status](https://readthedocs.org/projects/fedlab/badge/?version=latest)](https://fedlab.readthedocs.io/en/latest/?badge=latest) [![License](https://img.shields.io/github/license/SMILELab-FL/FedLab)](https://opensource.org/licenses/Apache-2.0) [![arXiv](https://img.shields.io/badge/arXiv-2107.11621-red.svg)](https://arxiv.org/abs/2107.11621) [![GH Actions Tests](https://github.com/SMILELab-FL/FedLab/actions/workflows/CI.yml/badge.svg)](https://github.com/SMILELab-FL/FedLab/actions) [![codecov](https://codecov.io/gh/SMILELab-FL/FedLab/branch/main/graph/badge.svg?token=4HHB5JCSC6)](https://codecov.io/gh/SMILELab-FL/FedLab)
+[![Documentation Status](https://readthedocs.org/projects/fedlab/badge/?version=latest)](https://fedlab.readthedocs.io/en/latest/?badge=latest) [![License](https://img.shields.io/github/license/SMILELab-FL/FedLab)](https://opensource.org/licenses/Apache-2.0) [![arXiv](https://img.shields.io/badge/arXiv-2107.11621-red.svg)](https://arxiv.org/abs/2107.11621) [![GH Actions Tests](https://github.com/SMILELab-FL/FedLab/actions/workflows/CI.yml/badge.svg)](https://github.com/SMILELab-FL/FedLab/actions) [![codecov](https://codecov.io/gh/SMILELab-FL/FedLab/branch/v1.0/graph/badge.svg?token=4HHB5JCSC6)](https://codecov.io/gh/SMILELab-FL/FedLab)
 
 
 _Read this in other languages: [English](README.md), [简体中文](README.zh-cn.md)._
diff --git a/README.zh-cn.md b/README.zh-cn.md
index b2607cc9..7dcaca71 100644
--- a/README.zh-cn.md
+++ b/README.zh-cn.md
@@ -2,7 +2,7 @@
 
 # FedLab: A Flexible Federated Learning Framework
 
-[![Documentation Status](https://readthedocs.org/projects/fedlab/badge/?version=latest)](https://fedlab.readthedocs.io/en/latest/?badge=latest)[![License](https://img.shields.io/github/license/SMILELab-FL/FedLab)](https://opensource.org/licenses/Apache-2.0) [![arXiv](https://img.shields.io/badge/arXiv-2107.11621-red.svg)](https://arxiv.org/abs/2107.11621) [![GH Actions Tests](https://github.com/SMILELab-FL/FedLab/actions/workflows/CI.yml/badge.svg)](https://github.com/SMILELab-FL/FedLab/actions) [![codecov](https://codecov.io/gh/SMILELab-FL/FedLab/branch/main/graph/badge.svg?token=4HHB5JCSC6)](https://codecov.io/gh/SMILELab-FL/FedLab)
+[![Documentation Status](https://readthedocs.org/projects/fedlab/badge/?version=latest)](https://fedlab.readthedocs.io/en/latest/?badge=latest)[![License](https://img.shields.io/github/license/SMILELab-FL/FedLab)](https://opensource.org/licenses/Apache-2.0) [![arXiv](https://img.shields.io/badge/arXiv-2107.11621-red.svg)](https://arxiv.org/abs/2107.11621) [![GH Actions Tests](https://github.com/SMILELab-FL/FedLab/actions/workflows/CI.yml/badge.svg)](https://github.com/SMILELab-FL/FedLab/actions) [![codecov](https://codecov.io/gh/SMILELab-FL/FedLab/branch/v1.0/graph/badge.svg?token=4HHB5JCSC6)](https://codecov.io/gh/SMILELab-FL/FedLab)
 
 _其他语言版本：[English](README.md), [简体中文](README.zh-cn.md)._