diff --git a/README.md b/README.md
index 32a544c..ca69c5c 100644
--- a/README.md
+++ b/README.md
@@ -18,12 +18,12 @@ import warnings
 warnings.filterwarnings("ignore")
 
 
-def set_seed(seed):
+def random_state(seed):
   torch.manual_seed(seed)
   if torch.cuda.is_available():
     torch.cuda.manual_seed_all(seed)
 
-set_seed(1234)
+random_state(1234)
 
 #Init models (make sure you init ONLY once if you integrate this to your code)
 parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False)