Update code
Merge pull request #2 from Perry5/local-branch

Update code