modified the return from both main and helper.

[patilabhay679]

It looks like there is no need to create a copy, we can replace the new elements inplace by mutating the contents of array.