Add test script to train classifier