Updated the code formatting