Added a warning class

(Really small change... needed to sync)