Add WASD support