add class logic

Signed-off-by: Andrea Corna <andrea.corna.ac.91@gmail.com>