* first draft * adding to code reference as well * docstrings * addressing comments * forgot opening line * docstring too