Add support for training/testing to the driver/model
When plotting trades, profitable will be green and non-profitable red