Updated code formatting