- [ ] Add docstrings to all functions. - [ ] Make sure everything has `Returns`. - [ ] Add attributes to docstring.