Updated to use arrow keys