Add test cases and script