Class name update ;)