add correct and wrong class to divs