refactor training clusters