[TASK] Added the functioning code that was added to the plugin instead of here