fixed issue 2
Merge pull request #13 from SophiaZhu314/issue_fix

fixed issue 2