Add comments in the code