updated function docstrings for HW1