Add unit tests

**Is your feature request related to a problem? Please describe.**
There is currently no unit testing in this project. 

**Describe the solution you'd like**
We would like to add a unit testing library to the projecct

**Describe alternatives you've considered**
The alternative is no testing.

**Additional context**
N/a
