Add a couple of functions