Small corrections in the code.