baseline script now uses all training data instead of just rost data
running tests and fixing minor bugs