Made a few more tweaks on the paper