Add a couple hooks