training fixed and readme improved