Adding box colors, need new matplotlib