Checklist:

• I have refactored my code out from notebooks/
• I have checked the code runs
• I have tested the code
• I have run pre-commit and addressed any issues not automatically fixed
• I have merged any new changes from dev
• I have documented the code
• Major functions have docstrings
• Appropriate information has been added to READMEs
• I have explained the feature in this PR or (better) in output/reports/
• I have requested a code review