Adding comments
minor fix in comments