Corrections in the code