small changes in code formatting