Can you add a description in the docstring of what number_of_segments is as well as number_of_axes? (seems to be missing from a past pull request).