add comments
little changes