Update traverse-with-class

Signed-off-by: Nicola Squartini <tensor5@gmail.com>