add docstring and comment
Add comments