Refator to more idiomatic style
- Reformat existing docstrings
- Add more docstrings for generated source code documentation