Improve test cases, add input parameters.
Improve test cases, add input parameters.