Add arrow key support