[TASK] Included functional tests to code coverage