Correcting some code formatting