submit a change to my branch

the original code cannot handle the array correctly
Merge pull request #1 from sxh2000251/master

the origianl code cannot array correctly