Creating a stock price prediction model