Works for the test dataset, now time to make sure everyone agrees.
After having some time to review the code, I'm a huge fan.