Change .getAttribute( 'class' ) => .className