code update
update code
Merge pull request #4 from ZiggyV/master

code updated