Implemented extended flow in test case generation.