Feature: classify.py train and test paths are command line arguments