Add python script to classify