local Clockwork = Clockwork;
local PLUGIN = PLUGIN;

function PLUGIN:PlayerButtonDown(player, key)
    if (not IsFirstTimePredicted()) then return end

    local activeWeapon = player:GetActiveWeapon():GetClass();

    if (activeWeapon == "cw_stunstick") and (key == 13) then

    Schema.HighVoltageButtonPanel = vgui.Create("cwHighVoltageButtonPanel");
    Schema.MediumVoltageButtonPanel = vgui.Create("cwMediumVoltageButtonPanel");
    Schema.LowVoltageButtonPanel = vgui.Create("cwLowVoltageButtonPanel");

    Schema.HighVoltageButtonPanel:MakePopup();
    Schema.MediumVoltageButtonPanel:MakePopup();
    Schema.LowVoltageButtonPanel:MakePopup();
    end;
end;

function PLUGIN:PlayerButtonUp(player, key)
    if (not IsFirstTimePredicted()) then return end

    local activeWeapon = player:GetActiveWeapon():GetClass();

    if (activeWeapon == "cw_stunstick") and (key == 13) then

    Schema.HighVoltageButtonPanel:Close();
    Schema.MediumVoltageButtonPanel:Close();
    Schema.LowVoltageButtonPanel:Close();
    end;
end;