Added post-processing to remove small steps