Add unit test to improve coverage

[SimonR99]

I think 50% coverage should be a decent amount considering it's a capstone project. I would be nice to also display it on the readme (tag)