clean up, make arrow keys work