Add some code to create test case