add unit and UI tests

**Is your feature request related to a problem? Please describe.**
Test coverage of the project is around 1%. 

**Describe the solution you'd like**
To improve detection of bugs and maintanability, tests should be implemented. (Unit + UI)
