Fixed cloaks and shields

diff --git a/code.lua b/code.lua
index c0e5f41..8192377 100644
--- a/code.lua
+++ b/code.lua
@@ -41,7 +41,7 @@ end
 local categoryMap = {
 	"INVTYPE_HEAD", --"Head",
 	"INVTYPE_SHOULDER", --"Shoulder",
-	"INVTYPE_BACK", --"Back",
+	"INVTYPE_CLOAK", --"Back",
 	"INVTYPE_CHEST", --"Chest",
 	"INVTYPE_SHIRT", --"Shirt",
 	"INVTYPE_TABARD", --"Tabard",
@@ -56,7 +56,7 @@ local categoryMap = {
 	"One-Handed Maces",
 	"Daggers",
 	"Fist Weapons",
-	"Shields",
+	"INVTYPE_SHIELD", --"Shields",
 	"Held In Off-hand",
 	"Two-Handed Axes",
 	"Two-Handed Swords",