Add POJ3844 code

Fix test code