modified the previous basic training ipynbs