small changes, fixed comment typo