Add plugin functionality