Fix sample code
Merge pull request #4 from SpyMaster356/master

Fix sample code