Update plugin.py

Bug fixes