Updated comments

For #18