Updated rectify step