diff --git a/train/dataset.py b/train/dataset.py
index 6da52cf..f41d051 100644
--- a/train/dataset.py
+++ b/train/dataset.py
@@ -1,3 +1,9 @@
+'''
+
+    Please implement a Dataset which inherit the PyTorch Dataset class.
+    So that our trainer can load the data from /dataset_dir
+
+'''
 import os
 import numpy as np
 from torch.utils.data import Dataset, DataLoader
diff --git a/train/model.py b/train/model.py
index a658c6d..1a01104 100644
--- a/train/model.py
+++ b/train/model.py
@@ -1,3 +1,8 @@
+'''
+
+    Please Implement your model here.
+
+'''
 from torch import nn
 import torch.nn.functional as F