added an option to use arrow keys