now it works with arrow keys