Update for direction control