Only for review sake

Instead of this way of proving the program works, I would have liked to see a unittest instead testing the class/methods