added some code. Plugin broke. Fix it tomorrow