Create unit tests for each functionality of the package - [ ] #7 - [ ] #8
Create unit tests for each functionality of the package