Some refactoring as suggested in code review feedback
Fix the trailing comments for variable declaration and function/arrow expressions
Emit space before emitting trailing comments
Start leading comments on new line if the comment starts on different line as node