Updating test class
Updating test case