Added classification module

Signed-off-by: Archit Shukla <architshukla2@gmail.com>