working on docstring now