📝 Add missing docstrings

### Related problem

When I linted the codebase in #40, I ignored the missing docstrings

### Proposed feature

Add all the missing docstrings, at least on public modules / functions / classes / methods

### Acceptance criteria

- [ ] no missing docstrings!

### Alternatives

_No response_

### Additional context

_No response_