Test case logic added