First of all, I noticed that you are not returning an array like what the question asked for. This line, return i*a, returns the first element of arr multiplied by a.