Updated the code formatting

Updated the code formatting