Update for final paper

Update for final paper