Reverted back and modified the problem.md to describe the expected output better