Add set array function