separated test and train

do not need to train everytime. No other changes in code or db