added functionality for full classification
added functionality for full classification