Use weapon classes

Move from colors to actual weapons icons and classes.