Updated the wrapper to handle two directions at once