Added classification, but there is a bug - an exception is falling