Add docstrings in the codebase

[anubhavsinha98]

The docs strings should be as follow:

def fun(a,b): '''what fun does Args: a:type:use b:type:use Return: c:type:use ''' return c