Changed className to attr('class')