Add basic balance plugin logic