added arrow key controls as well