[Task] Added basic unit test (wip)