Trying to fix arrow key problem