working on training refinement