Updated the code to comply with the project's code style.