add usage to train; add test option to train