Added back array return. whoops