Minor changes in code formatting