Problem 2 - correct return type for sumArray