Improve comment formatting