Add hooks for post before and after