submit a change to my branch

the original code cannot handle the array correctly