Support using arrow keys in play mode