Improvement Style
Add warning class