Yes I see that, maybe the code problems readme needs more definition and test cases.

I was merely putting a solution out there to start from as a talking point, which has worked!

Would be great if the pull request is merged, then you can open a new pull request with your update. Or alternatively you can work on an push to my branch.