added trainging adn testing for all data when the simulation is run