Improve unit tests coverage

We need to improve unit tests coverage in order to reach the value close to 100%

How to:

1. Analyze existing code base using the `coverage` package 
2. Review report generated by the `coverage` package and find the code not covered by tests
3. Prepare unit tests
4. Go to step 1 and  repeat action unit we cover almost 100% of the code base