Few lines added in TestCases