Use proper Swift Array type instead of NSArray