Changed the output of the problem