And could use the same function here