changed body removeClass().addClass() to attr 'class'