!! super update: -update at new version of library -now train.ipynb work