Added second part of the code
Merge pull request #4 from leonpalafox/master

Added second part of the code