Update comment to reflect changes