Update controls to also use WASD