Update classify.c

Updated main