Отличная идея использовать классы, чтобы выделить элементы.
Однако, исполнение несколько хромает:
Не забывайте, что у браузера уже может быть родной метод document.getElementsByClassName, который работает на два порядка шустрее, чем пребор документа. А может и не быть.
Не забывайте, что у элемента может быть несколько классов. Так что проверка
element.className == class
верна далеко не всегда. Лучше использовать
var classRegExp = new RegExp("(^|\\s)+" + class + "(\\s|$)+");
classRegExp.test(element.className)