Fixed the major bug in price prediction