Add some functionality to books controller