Add unit tests

[CleanCut]

We should add unit tests covering most of our logic, though we probably don't need any for the argument parsing giving it's a pretty straightforward usage of clap.