Added arrow key control