[Please explain what problem this PR is fixing or what feature it is adding.]
- PR has descriptive title explaining its purpose
- All merge conflicts are resolved (you may need to do this after creating the PR)
- Tests run successfully (see README for instructions)
- Function and class documentation is updated
[Please provide code that shows improved behaviour on this PR compared to on main. Consider also adding relevant tests to the tests folder, if applicable.]