Small corrections

We want our for loop to return an array of *all* int values, not
just the first one