changed x.train 40 to 91