Fixed code. Added plugin