Small fix to code to match its description.
Small fix to code indentation.
Merge pull request #349 from namin/patch-2

Small fix to code to match its description.