Update code formatting