Modify input code to support control sticks