updated test to take array