updates to plugin

Ok I think I did this right, I hope.