Add tests to the project

**Is your feature request related to a problem? Please describe.**
It a good point to apply left shift testing strategy

**Describe the solution you'd like**
Add tests for the project

**Describe alternatives you've considered**

**Additional context**
If you need a help, I could do this)
