fix issue 45 update
Merge pull request #47 from ywang19/FIX_ISSUE_45

fix issue 45 update