Updating for new class group