Added class to problem div.