added default to probability test case
exported count to train_models
Removed start of string test/train bug