Integrating labels and changes to classifier script