code correction
Merge pull request #125 from cwickham/patch-1

code correction