Simplify comment alignment logic