Correct code formatting issues