Correct version of the code