Update paper
Fix figure