Add arrow key movement functionality