Add plugin logic