adding changes to classifier script