Add changes in the paper