Added test class
Merge pull request #2 from ag-csw/master

Added test class