Working on training refinement stages