[Feature] Add more unit tests #18
Description
Is your feature request related to a problem? Please describe.
More unit tests are desired to ensure that each part is working as expected.
Describe the feature you'd like.
Unit tests added to each function.
Additional context.
This requires some refactoring to make scripts less stateful and more stateless (Issue #25) to happen first
How can this issue be closed?
Unit tests are added that cover all (or nearly all) functions