Prepare for optimization
Prepare for optimization
Add test case