First complete the following tasks: - [ ] API documentation (NumPy style docstrings) - [ ] adding tests using the pytest framework - [ ] #23
First complete the following tasks: