Add displacement task support