Small changes in code.