add test for returned array