Can you change the parameter name to `f` and also update the doc string which currently refers to the parameter as `i`?