Add move_unit callback to world2