Classification done but only one feature is extracted, to be improved