improve classification

Signed-off-by: Sébastien Boisvert <sebastien.boisvert.3@ulaval.ca>