diff --git a/README.md b/README.md
index d78f936..640a50b 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,4 @@
-# DeepTIMe: Deep Time-Index Meta-Learning for Non-stationary Forecasting
+# DeepTIMe: Deep Time-Index Meta-Learning for Non-Stationary Time-Series Forecasting
 
 <p align="center">
 <img src=".\pics\deeptime.png" width = "700" alt="" align=center />
@@ -84,4 +84,13 @@ authors for open-sourcing their work.
 
 * https://github.com/ElementAI/N-BEATS
 * https://github.com/zhouhaoyi/Informer2020
-* https://github.com/thuml/Autoformer
\ No newline at end of file
+* https://github.com/thuml/Autoformer
+
+## Citation
+
+Please consider citing if you find this code useful to your research.
+<pre>@article{woo2022deeptime,
+    title={DeepTIMe: Deep Time-Index Meta-Learning for Non-Stationary Time-Series Forecasting},
+    author={Gerald Woo and Chenghao Liu and Doyen Sahoo and Akshat Kumar and Steven C. H. Hoi},
+    year={2022},
+}</pre>
\ No newline at end of file
diff --git a/models/DeepTIMe.py b/models/DeepTIMe.py
index e08a490..327f371 100644
--- a/models/DeepTIMe.py
+++ b/models/DeepTIMe.py
@@ -1,3 +1,8 @@
+ # Copyright (c) 2022, salesforce.com, inc.
+ # All rights reserved.
+ # SPDX-License-Identifier: BSD-3-Clause
+ # For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+
 from typing import Optional
 
 import gin
diff --git a/models/modules/feature_transforms.py b/models/modules/feature_transforms.py
index d183838..dcfcdf6 100644
--- a/models/modules/feature_transforms.py
+++ b/models/modules/feature_transforms.py
@@ -1,3 +1,8 @@
+# Copyright (c) 2022, salesforce.com, inc.
+# All rights reserved.
+# SPDX-License-Identifier: BSD-3-Clause
+# For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+
 from typing import List, Optional
 
 import math
diff --git a/models/modules/inr.py b/models/modules/inr.py
index dde2b99..d0b65c5 100644
--- a/models/modules/inr.py
+++ b/models/modules/inr.py
@@ -1,3 +1,8 @@
+# Copyright (c) 2022, salesforce.com, inc.
+# All rights reserved.
+# SPDX-License-Identifier: BSD-3-Clause
+# For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+
 from typing import Optional
 
 import torch
diff --git a/models/modules/regressors.py b/models/modules/regressors.py
index a5615e7..68040b4 100644
--- a/models/modules/regressors.py
+++ b/models/modules/regressors.py
@@ -1,3 +1,8 @@
+# Copyright (c) 2022, salesforce.com, inc.
+# All rights reserved.
+# SPDX-License-Identifier: BSD-3-Clause
+# For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+
 from typing import Optional
 
 import torch