Code comments

Add small comments about the code