Addingtrain/predict  functionality to the command line interface