Implement unit tests for the package

[andybeet]

Is your feature request related to a problem? Please describe.

No problem.
Improvement: Implement unit tests for the package to ensure functions dont break