Added a test for the returning of the array