Merge pull request #1 from nlsudheer/code

Code
refactored the code
TestNG