Closes #50

Pull request checklist

Please check if your PR fulfills the following requirements:

• I have read the CONTRIBUTING doc.
• Tests for the changes have been added (for bug fixes / features).
• Docs have been reviewed and added / updated if needed (for bug fixes / features).
• Build (dotnet build) was run locally and any changes were pushed for failures / warnings.
• Tests (dotnet test) has passed locally and any fixes were made for failures.

Pull request type

Please check the type of change your PR introduces:

• Bugfix
• Feature
• Code style update (formatting, renaming)
• Refactoring (no functional changes, no api changes)
• Build related changes
• Documentation content changes
• Other (please describe):

What is the current behavior?

Tests coverage < 50%

What is the new behavior?

Tests coverage is 70%

Does this introduce a breaking change?

• Yes
• No

Other information

N/A