Unit tests should be added

**Is your feature request related to a problem? Please describe.**
Unit tests should be added to ensure nothing breaks

**Describe the solution you'd like**
Unit tests in Jest

**Describe alternatives you've considered**
Manual testing

**Additional context**
None
