Update with correct output