Correct solution