Return correct arrays. Fixes #10

Hopefully...
Always return an array, no matter what.