exercise 1, day 3... extending homework.
I'd already done it, but I had to rename and add the tests for it