Minor changes in code.