Cannot train properly...
 training fixed and readme improved