How to manipulate arrays in Swift?