Updated some code formatting