Added test code
Merge pull request #1 from davidaj3/test

Added test code