refactored the code
TestNG
Merge pull request #2 from nlsudheer/code

Code