Add arrow key and WASD movement functionality