Добавил возможность классификации. Работает не очень