Fix comment
Work when input ends with whitespaces