Fix comment
Fix whitespace code alignment