Wrote in one very simple unittest and wrote in the simplest possible solution.