[Feature Request] Write Unit Tests

**Is your feature request related to a problem? Please describe.**
Related to #17 as well. Feel like this should just be its own issue now that error checking/handling should mostly be completed and functional.

**Describe the solution you'd like**
Pytest tests that offer good coverage and testing of the logic in the script. Try to find edge cases that may be affected by future updates.