fix: update docs to match code example

[becelli-rubcube]

Description

Related Issue

Motivation and Context

How Has This Been Tested?

Types of changes

• [x] Updated docs / Refactor code / Added a tests case (non-breaking change)
• [ ] Bug fix (non-breaking change which fixes an issue)
• [ ] New feature (non-breaking change which adds functionality)
• [ ] Breaking change (fix or feature that would cause existing functionality to change)

Checklist:

• [x] My code follows the code style of this project.
• [x] My change requires a change to the documentation.
• [x] I have updated the documentation accordingly.
• [x] I have read the CONTRIBUTING document.
• [x] I have added tests to cover my changes.
• [x] All new and existing tests passed.
• [ ] I have updated the changelog.