Updated the widget template to take the right classes for appropriate styles