Add element argument to getRateCallback()

This can be useful if you want to customize the rate policy on a per-element basis.