Add POJ4054 code

Code not accepted by poj.org yet, possiblely WA
Improved test driver