Fixed so that highlights work for arrow keys

try not to use arraylists :)