Add rest hook support