Ruby - add array handling