minor changes to add comments