Adding hooks for WordPress