diff --git a/README.md b/README.md
index 0d560a5..b58a51c 100644
--- a/README.md
+++ b/README.md
@@ -40,11 +40,11 @@ pip install -r requirement.txt
 <th valign="bottom">ViT-Large</th>
 <!-- TABLE BODY -->
 <tr><td align="left">Colour fundus image</td>
-<td align="center"><a href="https://dl.fbaipublicfiles.com/mae/pretrain/mae_pretrain_vit_large.pth">download</a></td>
+<td align="center"><a href="https://drive.google.com/file/d/1l62zbWUFTlp214SvK6eMwPQZAzcwoeBE/view?usp=sharing">download</a></td>
 </tr>
 <!-- TABLE BODY -->
 <tr><td align="left">OCT</td>
-<td align="center"><a href="https://dl.fbaipublicfiles.com/mae/pretrain/mae_pretrain_vit_large.pth">download</a></td>
+<td align="center"><a href="https://drive.google.com/file/d/1m6s7QYkjyjJDlpEuXm7Xp3PmjN-elfW2/view?usp=sharing">download</a></td>
 </tr>
 </tbody></table>
 
diff --git a/engine_finetune.py b/engine_finetune.py
index f501738..54e7558 100644
--- a/engine_finetune.py
+++ b/engine_finetune.py
@@ -1,12 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# DeiT: https://github.com/facebookresearch/deit
-# BEiT: https://github.com/microsoft/unilm/tree/master/beit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import math
diff --git a/main_finetune.py b/main_finetune.py
index 6c174d4..42c5330 100644
--- a/main_finetune.py
+++ b/main_finetune.py
@@ -1,12 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# DeiT: https://github.com/facebookresearch/deit
-# BEiT: https://github.com/microsoft/unilm/tree/master/beit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import argparse
@@ -203,11 +197,6 @@ def main(args):
         else:
             sampler_test = torch.utils.data.SequentialSampler(dataset_test)
             
-            
-    else:
-        sampler_train = torch.utils.data.RandomSampler(dataset_train)
-        sampler_val = torch.utils.data.SequentialSampler(dataset_val)
-        sampler_test = torch.utils.data.SequentialSampler(dataset_test)
 
     if global_rank == 0 and args.log_dir is not None and not args.eval:
         os.makedirs(args.log_dir, exist_ok=True)
diff --git a/models_vit.py b/models_vit.py
index 75cc90a..d7334c3 100644
--- a/models_vit.py
+++ b/models_vit.py
@@ -1,12 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# timm: https://github.com/rwightman/pytorch-image-models/tree/master/timm
-# DeiT: https://github.com/facebookresearch/deit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 from functools import partial
diff --git a/util/datasets.py b/util/datasets.py
index f552c83..0e58750 100644
--- a/util/datasets.py
+++ b/util/datasets.py
@@ -1,15 +1,9 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# DeiT: https://github.com/facebookresearch/deit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import os
-import PIL
 from torchvision import datasets, transforms
 from timm.data import create_transform
 from timm.data.constants import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
diff --git a/util/lr_decay.py b/util/lr_decay.py
index 7fa11f1..c3b4eb5 100644
--- a/util/lr_decay.py
+++ b/util/lr_decay.py
@@ -1,12 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# ELECTRA https://github.com/google-research/electra
-# BEiT: https://github.com/microsoft/unilm/tree/master/beit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import json
diff --git a/util/lr_sched.py b/util/lr_sched.py
index 4cb682b..7a3a107 100644
--- a/util/lr_sched.py
+++ b/util/lr_sched.py
@@ -1,8 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
+# Partly revised by YZ @UCL&Moorfields
+# --------------------------------------------------------
 
 import math
 
diff --git a/util/misc.py b/util/misc.py
index 50123b1..a79d696 100644
--- a/util/misc.py
+++ b/util/misc.py
@@ -1,12 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# References:
-# DeiT: https://github.com/facebookresearch/deit
-# BEiT: https://github.com/microsoft/unilm/tree/master/beit
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import builtins
diff --git a/util/pos_embed.py b/util/pos_embed.py
index 6acf8bd..11f7128 100644
--- a/util/pos_embed.py
+++ b/util/pos_embed.py
@@ -1,10 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
-
-# This source code is licensed under the license found in the
-# LICENSE file in the root directory of this source tree.
-# --------------------------------------------------------
-# Position embedding utils
+# Partly revised by YZ @UCL&Moorfields
 # --------------------------------------------------------
 
 import numpy as np