Adding hooks for customization