- [x] Add missing docstrings for steps - [x] Adding missing docstrings for modules - [ ] Improve internal documentation