want to add

- Bring back the old training system

- If not, at least add a repeat button