Update plugin.py

Small fixes, hopefully will work