update step to take an array