**Task** * Add unit testing to the package, [help docs here](https://doc.rust-lang.org/rust-by-example/testing/unit_testing.html).