Update

Added a couple lines of code
Merge pull request #2 from BethTfiloh/master

Update