Minor changes to code formatting