Added warning class and integrated with action