improve test cases and add solution code