Refactor—minor comment update