Add after callback function to plugin