* [TASK] Add unit tests and get 100% code coverage * [BUGFIX] Fix bugs that appear during test writing