Change output to array instead of string