Fix minor code formatting