Add comments in code