working now, just fixing up the array