Update body class after load