Added test case code