Add Unit Tests

[okrohan]

[agentfarmx]

👨🏻‍🌾 The agent is at work!
PR coming – reply to the agent in the PR.
(this will take 15 minutes max)

[agentfarmx]

PR #7 ready for review.

Let's continue iterating there.