Updating function definition

Updates to Swift required this change in API.