add functionality.
Merge pull request #4 from vlee90/new-branch

add functionality.