Add direction arg