added classification feature