Slightly tidied up how the trainer position handles keys