I let a small local model write the refactor tests I was too lazy to write

This is a submission for the Hacktoberfest Open-Source AI Challenge Week 3: Slow and Steady ...