Add two methods to make it act a little more like activerecord