Fix/review 414

[mprzypasniak99]

Description

Fixes after review from PR #414

Motivation and Context

#414

Types of changes

• 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 not work as expected)

Checklist:

• My code follows the code style of this project.
• My change requires a change to the documentation.
• Automated functional tests have been added or modified to cover my changes (if applicable)
• I have updated the documentation accordingly.

---

I hereby agree to the terms of the Cogboard Contributor License Agreement.