Small changes in the code.