update the code for new class name.