First semi-working prototype for classification