add new test for functions returning arrays (passes at the mo)