We should further clarify docstring conventions, especially for optional arguments and array shapes. Type hints will also help with this.