made change for code review feedback: add comments