Some changes in test and package
Added the final array and return object