updated the correct output