Added stock price to train trading strategy