Address Ben's feedback from PR #100

Fixed comments.