Fix some small output problem

The output now matches the problem description