Add direction control